package com.itheima.controller;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.itheima.Dto.DishDto;
import com.itheima.config.R;
import com.itheima.pojo.Category;
import com.itheima.pojo.Dish;
import com.itheima.pojo.DishFlavor;
import com.itheima.service.CategoryService;
import com.itheima.service.DishFlavorService;
import com.itheima.service.DishService;
import org.apache.commons.lang.StringUtils;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.web.bind.annotation.*;
import java.util.ArrayList;
import java.util.List;
import java.util.Set;

/**
 * 菜品管理
 */
@RestController
@RequestMapping("/dish")
public class DishController {

    @Autowired
    private DishService dishService;

    @Autowired
    private CategoryService categoryService;

    @Autowired
    private DishFlavorService dishFlavorService;

    @Autowired
    private RedisTemplate redisTemplate;

    /**
     * 分页查询菜品数据
     * @param page
     * @param pageSize
     * @param name 模糊条件
     * @return
     */
    @GetMapping("/page")
    public R<Page> page(Integer page,Integer pageSize,String name){

        // 分页对象
        Page<Dish> dishPage = new Page<>(page,pageSize);

        // 返回的数据对象
        Page<DishDto> dtoPage = new Page<>();

        // 根据 菜品名称 模糊查询
        LambdaQueryWrapper<Dish> lambdaQueryWrapper = new LambdaQueryWrapper<>();
        lambdaQueryWrapper.like(StringUtils.isNotEmpty(name),Dish::getName,name);

        // 分页查询 附带模糊条件
        dishService.page(dishPage,lambdaQueryWrapper);

        // 获取分页查询后 所有字段数据集合
        List<Dish> records = dishPage.getRecords();

        ArrayList<DishDto> dishDtos = new ArrayList<>();

        // 循环每行数据
        for (Dish record : records) {

            // 创建这个类 去接收赋的值  没有的字段 会赋值给父类
            DishDto dishDto = new DishDto();

            // 获取每次循环 两个表相关联的ID字段
            Long categoryId = record.getCategoryId();

            //获取到 分类表 的ID
            Category byId = categoryService.getById(categoryId);

            // 获取到 分类表的  分类名称
            String byIdName = byId.getName();

            // 获取到的分类名称 赋值给  新对象
            dishDto.setCategoryName(byIdName);

            // 把一个有数据的对象的数据 赋值  给 另外一个对象    `
            BeanUtils.copyProperties(record,dishDto);

            //把赋值后的对象 放入集合 要不然 请求关闭后 无法获取到了
            dishDtos.add(dishDto);

        }

        // set dishDtos表
        dtoPage.setRecords(dishDtos);
        // 获取 数据的总数 分页显示总条数
        dtoPage.setTotal(dishPage.getTotal());

        return R.success(dtoPage);
    }

    /**
     * 新建菜品
     *
     *
     * @param dishDto
     * @return
     */
    @PostMapping    // dishDto 包含两个表所有属性
    public R<String> save(@RequestBody DishDto dishDto){

        // 直接把属性添加到Dish表  这时候 口味属性还没赋值
        dishService.save(dishDto);


        String key = "dish"+dishDto.getCategoryId();

        // 新增了菜品所以 要把Redis以前村的数据删掉，这样下次再重新缓存，就保证一致性
        redisTemplate.delete(key);


        // 口味集合
        List<DishFlavor> flavorslist = dishDto.getFlavors();

        // 口味表 的ID没有赋值 从DishDto中获取ID
        Long id = dishDto.getId();

        // 判断是否选择口味
        if (flavorslist != null && flavorslist.size() > 0){

            // 循环每个口味
            for (DishFlavor dishFlavor : flavorslist) {
                // 对每个口味 进行ID赋值
                dishFlavor.setDishId(id);
            }
        }
        return R.success("添加成功");
    }

    /**
     * 根据ID修改  回显
     * @param id
     * @return
     */
    @GetMapping("/{id}")
    public R<DishDto> echo(@PathVariable Long id){

        // 根据ID 查询菜品表的一条数据
        Dish byId = dishService.getById(id);

        LambdaQueryWrapper<DishFlavor> queryWrapper = new LambdaQueryWrapper<>();
        queryWrapper.eq(DishFlavor::getDishId,id);

        // 根据ID 查询口味表getDishId 字段 一个人会有多个口味
        List<DishFlavor> list = dishFlavorService.list(queryWrapper);

        DishDto dishDto = new DishDto();

        BeanUtils.copyProperties(byId,dishDto);

        //  给dishDto赋 口味
        dishDto.setFlavors(list);

        // 把封装所有数据的对象返回
        return R.success(dishDto);
    }

    /**
     *  菜品管理中的修改
     * @param dishDto
     * @return
     */
    @PutMapping
    public R<String> update(@RequestBody DishDto dishDto){

        // updateById 会自动根据传递的对象找里面的ID 进行数据更新
        dishService.updateById(dishDto);

        // 这里修改 可以修改菜品分类，所以要获取全部的dish 去删除
        Set keys = redisTemplate.keys("dish_*");
        redisTemplate.delete(keys);

        // 获取删除的 ID条件
        LambdaQueryWrapper<DishFlavor> queryWrapper = new LambdaQueryWrapper<>();
        queryWrapper.eq(DishFlavor::getDishId,dishDto.getId());

        // 根据ID删除所有口味 再进行添加
        dishFlavorService.remove(queryWrapper);


        // dishDto 获取Flavors集合 里面包含属性
        List<DishFlavor> flavors = dishDto.getFlavors();

        // 循环获取Flavors集合  每次循环里面一个集合
        for (DishFlavor flavor : flavors) {

            // 由于已经删除数据库中的数据，所以前端传入数据的时候没有dishId, 但是新增口味的时候dishDto 中一定会有ID，所有通过这个ID赋值给dishId
            flavor.setDishId(dishDto.getId());

            // 循环添加数据，每次添加这个字段中的一个集合的数据
            dishFlavorService.save(flavor);

        }





//        flavors = flavors.stream().map((item) -> {
//            item.setDishId(dishDto.getId());
//            return item;
//        }).collect(Collectors.toList());


        return R.success("修改成功");
    }

    /**
     * 逻辑删除 删除后页面看不到条数，isDeleted字段变成1
     * @param ids
     * @return
     */
    @Transactional
    @DeleteMapping
    public R<String> delete(@RequestParam List<Long> ids){

       LambdaQueryWrapper<Dish> queryWrapper = new LambdaQueryWrapper<>();
       queryWrapper.in(Dish::getCategoryId, ids);

        LambdaQueryWrapper<DishFlavor> wrapper = new LambdaQueryWrapper<>();
        wrapper.in(DishFlavor::getDishId,ids);

        // 菜品 逻辑删除
        dishService.remove(queryWrapper);

        // 口味直接删掉
        dishFlavorService.remove(wrapper);

       return R.success("删除成功");
//
//        for (Long id : ids) {
//            dishService.removeById(id);
//        }
    }

    /**
     *  0 停售   1起售  状态按钮
     * @param id
     * @param ids
     * @return
     */
    @PostMapping("/status/{id}")
    public R<String> status(@PathVariable Integer id,@RequestParam List<Long> ids){

        // 循环提供的id集合 获取到每个id
        for (Long idslist : ids) {

            // 获取对象好封装
            Dish dish = new Dish();

            // 这里id是两个表相关联的ID  修改根据这个id
            dish.setId(idslist);

            // 前端传的id 0和1 就是状态 直接赋值
            dish.setStatus(id);

            // 这里根据上面的id 和已经修改好的状态 去更新
            dishService.updateById(dish);
        }
        return R.success("操作成功");

    }

    /**
     * 套餐管理  新增中的添加菜品的查询
     * @param categoryId
     * @return
     */
    @GetMapping("/list")
    public R<List<DishDto>> list(Long categoryId){

        // 定义key   这个里拼接ID 是因为每个菜类的ID 都不一样 所以用来区分类的这样
        String key = "dish"+categoryId;
        // 去Redis中查询是否有数据
        List<DishDto> redis = (List<DishDto>) redisTemplate.opsForValue().get(key);

        // 判断查询出的是否为空
        if (redis != null){
            return R.success(redis);
        }

        LambdaQueryWrapper<Dish> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(Dish::getCategoryId,categoryId);

        // 只要起售状态的菜品
        wrapper.eq(Dish::getStatus,1);

        // 根据categoryId菜品分类    可以查询多个菜
        List<Dish> list = dishService.list(wrapper);

        ArrayList<DishDto> arrayList = new ArrayList<>();

        for (Dish dish : list) {

            // 循环每个dish 获取他的ID  因为彩皮ID  关联   口味表的 dish_ID字段
            LambdaQueryWrapper<DishFlavor> dishFlavorWrapper = new LambdaQueryWrapper<>();
            dishFlavorWrapper.eq(DishFlavor::getDishId,dish.getId());

            // 通过上面的条件查询  一个菜品有多个口味 所以是List
            List<DishFlavor> flavors = dishFlavorService.list(dishFlavorWrapper);

            // 获取category 一条数据 这里获取ID 目的是后面通过ID 获取Name
            Category byId = categoryService.getById(categoryId);

            // 查询出来的数据 封装到 Dto中 好返回
            DishDto dishDto = new DishDto();
            dishDto.setName(byId.getName());
            dishDto.setFlavors(flavors);

            BeanUtils.copyProperties(dish,dishDto);
            arrayList.add(dishDto);
        }
        // 走到这里说明Redis中没有数据，所以要吧查询的数据添加到Redis
        redisTemplate.opsForValue().set(key,arrayList);

        return R.success(arrayList);
    }

}
